package cn.slipi.admin.common.utils;

import freemarker.template.Template;
import org.apache.commons.codec.Charsets;

import java.io.*;
import java.nio.charset.StandardCharsets;
import java.util.Map;
import java.util.Objects;

public class WordUtil {

    /**
     * 生成word文件
     *
     * @param dataMap      word中需要展示的动态数据，用map集合来保存
     * @param templatePath word模板名称，例如：test.ftl
     * @param destFile     文件生成的目标路径，例如：D:/wordFile/xxx.doc
     */
    @SuppressWarnings("unchecked")
    public static void createWord(Map dataMap, String templatePath, String destFile) {
        try {
            //获取模板
            Template template = new Template(null, new InputStreamReader(Objects.requireNonNull(WordUtil.class.getClassLoader().getResourceAsStream(templatePath)), Charsets.UTF_8), null);

            //输出文件
            File outFile = new File(destFile);

            //如果输出目标文件夹不存在，则创建
            if (!outFile.getParentFile().exists()) {
                outFile.getParentFile().mkdirs();
            }

            //将模板和数据模型合并生成文件
            Writer out = new BufferedWriter(new OutputStreamWriter(new FileOutputStream(outFile), StandardCharsets.UTF_8));

            //生成文件
            template.process(dataMap, out);

            //关闭流
            out.flush();
            out.close();
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
}